import pandas as pd


def calculate_correlation(excel_path, column1, column2):
    """
    从Excel文件中读取数据并计算两列之间的相关系数。

    参数:
    excel_path (str): Excel文件的路径。
    column1 (str): 第一列的名称。
    column2 (str): 第二列的名称。

    返回:
    float: 相关系数。
    """
    # 读取Excel文件
    df = pd.read_excel(excel_path)

    # 处理缺失值，这里我们选择删除含有缺失值的行
    df = df.dropna(subset=[column1, column2])

    # 计算并返回相关系数
    return df[column1].corr(df[column2])



if __name__ == '__main__':
    # excel_path = './combined_product_data.xlsx'
    # correlation = calculate_correlation(excel_path, '综合服务评分', '销售总量')
    # print(f"商品回购率与销售总量的相关系数为: {correlation}")

    # 读取Excel文件
    file_path = './combined_product_data.xlsx'
    df = pd.read_excel(file_path)

    # 去除末尾的"、"
    df = df.applymap(lambda x: x.rstrip('、') if isinstance(x, str) else x)

    # 另存为新的Excel文件
    output_file_path = './combined_product_data_2.xlsx'
    df.to_excel(output_file_path, index=False)


